package com.slf.demo.config;

import com.slf.demo.utils.SpringUtil;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

import java.util.HashMap;
import java.util.Map;

/**
 * @author slf
 * @date 2021/7/15
 */
public class DynamicDataSource extends AbstractRoutingDataSource {
    private static final ThreadLocal<String> dataSourceKey = ThreadLocal.withInitial(() -> "defaultDataSource");
    public static Map<Object, Object> dataSourceMap = new HashMap<>();

    static {
        dataSourceMap.put("defaultDataSource", SpringUtil.getBean("defaultDataSource"));
    }

    /**
     * 确定当前的查找键。 这通常用于检查线程绑定事务上下文。
     * 允许任意键。 返回的键需要匹配存储的查找键类型，由 super.resolveSpecifiedLookupKey(java.lang.Object) 方法解析。
     *
     * @return
     */
    @Override
    protected Object determineCurrentLookupKey() {
        return dataSourceKey.get();
    }

    /**
     * 设置当前线程数据源
     *
     * @param dataSource 数据源名称（需要在数据源配置中已配置）
     */
    public static void setDataSource(String dataSource) {
        dataSourceKey.set(dataSource);
    }

    /**
     * 获取当前线程数据源名称
     *
     * @return
     */
    public static String getDataSource() {
        return dataSourceKey.get();
    }

    /**
     * 移除当前数据源，返回到默认数据源
     */
    public static void clear() {
        dataSourceKey.remove();
    }

    /**
     * 刷新数据源配置
     */
    public static void refreshDataSource() {
        DynamicDataSource dynamicDataSource = (DynamicDataSource) SpringUtil.getBean("dataSource");
        dynamicDataSource.afterPropertiesSet();
    }

}
